-
Notifications
You must be signed in to change notification settings - Fork 33
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add inference API and task for all BioImage.IO Model Zoo models #338
Conversation
Codecov ReportAttention: Patch coverage is
Additional details and impacted files@@ Coverage Diff @@
## master #338 +/- ##
==========================================
- Coverage 65.25% 63.45% -1.81%
==========================================
Files 72 73 +1
Lines 5301 5442 +141
==========================================
- Hits 3459 3453 -6
- Misses 1842 1989 +147 ☔ View full report in Codecov by Sentry. |
for bioimageio.core blocked prediction
with broken types and returns
316aa91
to
cf0e69b
Compare
@qin-yu the ci seems to be stuck |
PyTest finishes without error on macOS and Linux for me. Checking the CI now. |
For the record, starting with Python 3.13, |
I rerun the CI for main branch and it got stuck as well:
|
Now not only axes with `.id` and `.size` attributes but also string axes such as 'bcyx' coming with `shape` at the same level. These are due to the legacy of bioimage.io and complexity of versioning. Tested with `pioneering-rhino`
469aa6c
to
fabae0a
Compare
Add inference API and task for all BioImage.IO Model Zoo models
PlantSeg v2 supports training of our models and inference of all BioImage.IO models. This PR turns v2.0.0aX to v2.0.0b0.
Main features
I integrated support for bioimageio.core inference with BioImage.IO Model Zoo models into the PlantSeg API, tasks, and widgets.
To ensure functionality, I conducted the following manual tests:
1. 2D Input (YX): Tested with the
pioneering-rhino
PlantSeg model, resulting in a 2D image output.2. 3D Input (ZYX): Tested with the
emotional-cricket
PlantSeg model, resulting in a 3D image output.3. 4D Input (CZYX): Tested with the
philosophical-panda
Cellpose model, resulting in a sequence of 4D image outputs.The bioimageio.core Python package has multiple versions, requiring careful handling of format compatibility and extensive validation. Ongoing development of the library introduces additional complexity, but I believe this implementation addresses most use cases and version-specific challenges.
Screenshot
Multiple outputs of different sizes from Cellpose network:
List of BioImage.IO Model Zoo models: